import java.util.HashSet;
import java.util.Set;

/**
 * 547. 省份数量
 * <p>
 * 有 n 个城市，其中一些彼此相连，另一些没有相连。如果城市 a 与城市 b 直接相连，且城市 b 与城市 c 直接相连，那么城市 a 与城市 c 间接相连。
 * <p>
 * 省份 是一组直接或间接相连的城市，组内不含其他没有相连的城市。
 * <p>
 * 给你一个 n x n 的矩阵 isConnected ，其中 isConnected[i][j] = 1 表示第 i 个城市和第 j 个城市直接相连，而 isConnected[i][j] = 0 表示二者不直接相连。
 * <p>
 * 返回矩阵中 省份 的数量。
 * <p>
 * 示例 1：
 * <p>
 * 输入：isConnected = [[1,1,0],[1,1,0],[0,0,1]]
 * 输出：2
 * <p>
 * 示例 2：
 * <p>
 * <p>
 * 输入：isConnected = [[1,0,0],[0,1,0],[0,0,1]]
 * 输出：3
 */
public class NumberOfProvinces {
    public static int findCircleNum(int[][] isConnected) {
        UnionFind unionFind = new UnionFind(isConnected.length);
        for (int i = 0; i < isConnected.length; i++) {
            for (int j = 0; j < isConnected[i].length; j++) {
                if (isConnected[i][j] == 1) {
                    unionFind.union(i, j);
                }
            }
        }
        return unionFind.getSize();
    }

    public static void main(String[] args) {
        System.out.println(findCircleNum(new int[][]{new int[]{1, 0, 0}, new int[]{0, 1, 0}, new int[]{0, 0, 1},}));
    }
}


class UnionFind {
    private final int[] size;
    private final int[] parent;
    private final int[] help;
    private int target;

    public UnionFind(int target) {
        parent = new int[target];
        size = new int[target];
        help = new int[target];
        this.target = target;
        for (int i = 0; i < target; i++) {
            parent[i] = i;
            size[i] = 1;
        }
    }

    public int findOrigin(int i) {
        int hi = 0;
        while (parent[i] != i) {
            help[hi++] = i;
            i = parent[i];
        }
        for (hi--; hi >= 0; hi--) {
            parent[help[hi]] = i;
        }
        return i;
    }

    public void union(int i, int j) {
        int oi = findOrigin(i);
        int oj = findOrigin(j);
        if (oi != oj) {
            int iMax = size[oj];
            int jMax = size[oi];
            if (iMax > jMax) {
                parent[oj] = oi;
            } else {
                parent[oi] = oj;
            }
            target--;
        }
    }

    public int getSize() {
        return this.target;
    }
}
